
import pandas as pd
import numpy as np

input_file = 'wuhan_price.csv'

# 读取CSV文件
try:
    df = pd.read_csv(input_file)

    # 将DataFrame随机打乱（可选，如果希望随机分配行）
    # df = df.sample(frac=1).reset_index(drop=True)

    # 将DataFrame分割成3个部分
    split_dfs = np.array_split(df, 3)

    # 将每个部分保存为新的CSV文件
    for i, part_df in enumerate(split_dfs):
        output_file = f'wuhan_price_part_{i+1}.csv'
        part_df.to_csv(output_file, index=False)
        print(f'已创建文件: {output_file}')

except FileNotFoundError:
    print(f"错误：找不到文件 '{input_file}'。请确保文件名正确并已上传。")